FROM uhub.service.ucloud.cn/eagle_nest/ubuntu16.04-cuda10.0-cudnn7.4-opencv4.1-snpe1.47-pytorch1.5-workspace

# 创建默认目录，训练过程中生成的模型文件、日志、图必须保存在这些固定目录下，训练完成后这些文件将被保存
RUN mkdir -p /project/train/src_repo && mkdir -p /project/train/result-graphs && mkdir -p /project/train/log && mkdir -p /project/train/models/2021511resnet_super

# 安装训练环境依赖端软件，请根据实际情况编写自己的代码
COPY ./ /project/train/src_repo/
# 创建默认目录，训练过程中生成的模型文件、日志、图必须保存在这些固定目录下，训练完成后这些文件将被保存
# RUN python3.6 -m pip install -r /project/train/src_repo/requirements.txt
RUN pip3.6 install torchvision
RUN pip3.6 install scikit-image
RUN pip3.6 install scikit-learn
RUN pip3.6 install efficientnet_pytorch

